package com.leetcode.algorithm.y19.m02;

/**
 * leetcode-cn.com
 * (done)977. 有序数组的平方
 * (done)985. 查询后的偶数和
 * @author: jie.deng
 * @time: 2019年2月24日 下午9:14:17
 */
public class MySolution0224 {
    /**
     * 977. 有序数组的平方
     * 
	 * 给定一个按非递减顺序排序的整数数组 A，返回每个数字的平方组成的新数组，要求也按非递减顺序排序。
	 * 
	 *  
	 * 
	 * 示例 1：
	 * 
	 * 输入：[-4,-1,0,3,10]
	 * 输出：[0,1,9,16,100]
	 * 示例 2：
	 * 
	 * 输入：[-7,-3,2,3,11]
	 * 输出：[4,9,9,49,121]
	 *  
	 * 
	 * 提示：
	 * 
	 * 1 <= A.length <= 10000
	 * -10000 <= A[i] <= 10000
	 * A 已按非递减顺序排序。
     * @param A
     * @return
     */
	public int[] sortedSquares(int[] A) {
		int len = A.length;
		int[] ret = new int[len];
		int size = 0;// ret数组当前size
		for (int i = 0; i < len; i++) {
			int idx = size;// 待插入的索引
			int num = A[i] * A[i];
			// 将num插入已排序的数组ret里
			for (int j = 0; j < size; j++) {
				if (ret[j] > num) {
					idx = j;
					// [j,idx-1]后移一位
					for (int k = size; k > j; k--) {
						ret[k] = ret[k - 1];
					}
					break;
				}
			}
			ret[idx] = num;
			size++;
		}
		return ret;
	}
	
	/**
	 * 985. 查询后的偶数和
	 * 
	 * 给出一个整数数组 A 和一个查询数组 queries。
	 * 
	 * 对于第 i 次查询，有 val = queries[i][0], index = queries[i][1]，我们会把 val 加到 A[index] 上。然后，第 i 次查询的答案是 A 中偶数值的和。
	 * 
	 * （此处给定的 index = queries[i][1] 是从 0 开始的索引，每次查询都会永久修改数组 A。）
	 * 
	 * 返回所有查询的答案。你的答案应当以数组 answer 给出，answer[i] 为第 i 次查询的答案。
	 * 
	 *  
	 * 
	 * 示例：
	 * 
	 * 输入：A = [1,2,3,4], queries = [[1,0],[-3,1],[-4,0],[2,3]]
	 * 输出：[8,6,2,4]
	 * 解释：
	 * 开始时，数组为 [1,2,3,4]。
	 * 将 1 加到 A[0] 上之后，数组为 [2,2,3,4]，偶数值之和为 2 + 2 + 4 = 8。
	 * 将 -3 加到 A[1] 上之后，数组为 [2,-1,3,4]，偶数值之和为 2 + 4 = 6。
	 * 将 -4 加到 A[0] 上之后，数组为 [-2,-1,3,4]，偶数值之和为 -2 + 4 = 2。
	 * 将 2 加到 A[3] 上之后，数组为 [-2,-1,3,6]，偶数值之和为 -2 + 6 = 4。
	 *  
	 * 
	 * 提示：
	 * 
	 * 1 <= A.length <= 10000
	 * -10000 <= A[i] <= 10000
	 * 1 <= queries.length <= 10000
	 * -10000 <= queries[i][0] <= 10000
	 * 0 <= queries[i][1] < A.length 
	 * @param A
	 * @param queries
	 * @return
	 */
	public int[] sumEvenAfterQueries(int[] A, int[][] queries) {
		int sum = 0; // 偶数和
		for (int i = 0; i < A.length; i++) {
			if (A[i] % 2 == 0) {
				sum += A[i];
			}
		}

		int len = queries.length;
		int[] answer = new int[len];
		for (int i = 0; i < len; i++) {
			int val = queries[i][0];
			int index = queries[i][1];
			if (A[index] % 2 == 0) {
				if (val % 2 == 0) {
					sum += val;
				} else {
					sum -= A[index];
				}
			} else {
				if (val % 2 != 0) {
					sum += A[index];
					sum += val;
				}
			}
			A[index] += val;
			answer[i] = sum;
		}
		return answer;
	}
}
